#!/bin/bash

print_help() {
    echo "使用方法: $0 [选项]"
    echo
    echo "选项:"
    echo "  --help          显示帮助信息"
    echo "  --input_dir         文件夹输入路径"
    echo "  --output_dir        文件夹输出路径"
    echo "  --label_path   标签文件位置"
    echo "  [其他选项]      脚本的其他选项"
    echo
    echo "示例:"
    echo "  $0 --help"
}
input_path=""  #文件夹输入路径
output_path=""  #文件夹输出路径
label_path=""  #标签文件位置
# 解析命令行参数
while [[ "$#" -gt 0 ]]; do
    case $1 in
        --help)
            print_help
            exit 0
            ;;
        --output)
            output_path="$2"
            shift
            ;;
        --input)
            input_path="$2"
            shift
            ;;
        --residual)
            label_path="$2"
            shift
            ;;
        *)
            # 可以在这里处理其他选项
            ;;
    esac
    shift  # 移过已经读取的参数
done


if [ ${output_path} = "" ]; then
    echo "输出路径为空, 默认输出路径等于输入路径"
    output_path=$input_path
fi
if [ ${output_path} = ${input_path} ]; then
    echo "输出路径等于输入路径, 原地进行后续操作"
else
    echo "输入输出路径不一样"
    echo "清空输出路径中所有内容, 所有输入路径中的 images和Annotations存储于该输出路径中"
    rm -rf $output_path
    mkdir $output_path
fi

# #测试读入参数
# echo "输出路径：$output_path"
# echo "输入路径：$input_path"

echo "开始划分 分割voc格式训练数据"
python myutil/make_dataset/labelme2Edgevoc.py --input_dir ${input_path} --output_dir ${output_path} --label_path ${label_path}
python myutil/make_dataset/generate_ImageSets.py --root_dir ${output_path}
echo "划分完成啦"
